from ai_assistant import AIAssistant

class ImageGenerationAssistant(AIAssistant):
    def __init__(self, id, name, version, is_online, ability_score, max_resolution):
        super().__init__(id, name, "图片生成", version, is_online, ability_score)
        self.max_resolution = max_resolution
    
    def execute_task(self, task_content):
        return f"[{self.get_name()}] 执行图片生成任务：\n任务内容：{task_content}\n生成结果：已生成符合描述的图片，分辨率：{self.max_resolution}"
    
    # Getter and setter for max_resolution
    def get_max_resolution(self):
        return self.max_resolution
    
    def set_max_resolution(self, max_resolution):
        self.max_resolution = max_resolution
    
    def __str__(self):
        return super().__str__() + f",最大分辨率{self.max_resolution}"